/*
 * Copyright (C) 2024 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.google.cloud.teleport.v2.source.reader.io.schema;

import static com.google.common.truth.Truth.assertThat;

import com.google.cloud.teleport.v2.source.reader.io.jdbc.JdbcSchemaReference;
import junit.framework.TestCase;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.junit.MockitoJUnitRunner;

/** Test class for {@link SourceSchema}. */
@RunWith(MockitoJUnitRunner.class)
public class SourceSchemaTest extends TestCase {

  @Test
  public void testJdbcSourceSchemaBuilds() {
    final String testDb = "test-db";
    final String testTable = "test-table";

    SourceSchema testSchema =
        SourceSchema.builder()
            .setSchemaReference(
                SourceSchemaReference.ofJdbc(
                    JdbcSchemaReference.builder().setDbName(testDb).build()))
            .addTableSchema(SchemaTestUtils.generateTestTableSchema(testTable))
            .build();
    assertThat(testSchema.schemaReference().jdbc().dbName()).isEqualTo(testDb);
    assertThat(testSchema.tableSchemas()).hasSize(1);
    assertThat(testSchema.tableSchemas().get(0).avroSchema().getFields()).hasSize(2);
  }
}
